# WORKDIR: bert-tagger
CODEDIR=${WORKDIR}/playground
MODEL=bert-base-cased
python ${CODEDIR}/train-bert.py ${LOCATION} \
  --output_dir ${WORKDIR}/train_output -f \
  --train_dir ${WORKDIR}/train_data_sample \
  --eval_dir ${WORKDIR}/eval_data_sample \
  --do_eval \
  --device_id 1 \
  --num_train_epochs -1 \
  --log_interval 50 \
  --eval_interval 0 \
  --save_interval 100 \
  --train_batch_size 8 \
  --update_cycle 4 \
  --bert_learning_rate 1e-5 \
  --learning_rate 2e-5 \
  --max_input_len 128 \
  --max_num_candidates 30 \
  --max_top_antecedents 5 \
  --max_num_relations 25 \
  --bert_model ${MODEL} \
  --model ${WORKDIR}/pretrained_models/${MODEL}/model \
  --tokenizer ${WORKDIR}/pretrained_models/${MODEL}/tokenizer
  
